from sklearn.linear_model import Ridge
import numpy as np

np.random.seed(10) # 设置随机数种子

X = np.array(np.random.randint(5, size=(10, 10)))
Y = np.array(np.random.randint(10, size=(10,1 )))

alpha = 0.5

def ridge_model(X, Y, alpha):
    """
    参数:
    X -- 自变量数据矩阵
    Y -- 因变量数据矩阵
    alpha -- lamda 参数

    返回:
    W -- 岭回归系数
    """

    ### 代码开始 ### (≈ 3 行代码)
    model = Ridge(alpha, fit_intercept=False)
    model.fit(X,Y)
    W = model.coef_
    ### 代码结束 ###

    return W

ridge_model(X, Y, alpha)